import unittest
def add(a,b):
    return a+b
def sub(a,b):
    return a-b
class mathtest(unittest.TestCase):
    def testadd(self):
        self.assertEqual(add(1,1),2)
    def testsub(self):
        self.assertEqual(sub(2,1),1)
unittest.main()
